package string;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 438. 找到字符串中所有字母异位词
 * <p>
 * 给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。
 * 不考虑答案输出的顺序。
 * <p>
 * 示例 1:
 * 输入: s = "cbaebabacd", p = "abc"
 * 输出: [0,6]
 * 解释:
 * 起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
 * 起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。
 * <p>
 * 示例 2:
 * 输入: s = "abab", p = "ab"
 * 输出: [0,1,2]
 * 解释:
 * 起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。
 * 起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。
 * 起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。
 */
public class Problem_438 {
    public static void main(String[] args) {
        System.out.println(new Problem_438().findAnagrams_1("cbaebabacd", "abc"));
        System.out.println(new Problem_438().findAnagrams_2("abab", "ab"));
    }

    /**
     * 方法一：滑动窗口
     * 在字符串 s 中构造一个长度与字符串 p 长度相同的滑动窗口，
     * 并在滑动过程中维护窗口中每种字母的数量；
     * 当窗口中每种字母的数量与字符串 p 中每种字母的数量相同时，则说明当前窗口为字符串 p 的异位词。
     *
     * @param s
     * @param p
     * @return
     */
    public List<Integer> findAnagrams_1(String s, String p) {
        List<Integer> result = new ArrayList<>();
        int sLength = s.length();
        int pLength = p.length();

        if (sLength < pLength) {
            return result;
        }

        // 记录对应下标字母的出现次数
        int[] pHash = new int[26];
        int[] sHash = new int[26];
        // 初始化 pHash 和 sHash
        for (int i = 0; i < pLength; i++) {
            pHash[p.charAt(i) - 'a']++;
            sHash[s.charAt(i) - 'a']++;
        }
        if (Arrays.equals(sHash, pHash)) {
            result.add(0);
        }

        for (int start = 1; start < sLength - pLength + 1; start++) {
            sHash[s.charAt(start - 1) - 'a']--;
            sHash[s.charAt(start + pLength - 1) - 'a']++;
            if (Arrays.equals(sHash, pHash)) {
                result.add(start);
            }
        }

        return result;
    }

    public List<Integer> findAnagrams_2(String s, String p) {
        List<Integer> result = new ArrayList();
        int targetLength = p.length();

        char[] pChars = p.toCharArray();
        Arrays.sort(pChars);

        for (int i = 0; i < s.length() - targetLength + 1; i++) {
            char[] subChars = s.substring(i, i + targetLength).toCharArray();
            Arrays.sort(subChars);
            if (Arrays.equals(pChars, subChars)) {
                result.add(i);
            }
        }
        return result;
    }

}
